from huawei_inference_engine import HuaweiInferenceEngine
from baidu_inference_engine import BaiduInferenceEngine
from base_inference_engines import BaseInferenceEngine

def inference_engine_factory(device_type: str) -> BaseInferenceEngine:
    if device_type == "BAIDU":
        return BaiduInferenceEngine()
    elif device_type == "HUAWEI":
        return HuaweiInferenceEngine()
    else:
        raise ValueError(f"Unsupported device type: {device_type}")
